ReToken: One Token to Improve Vision-Language Models for Visual Retrieval
ReToken:用一个Token提升视觉语言模型的长视觉检索
🧭 多模态视觉检索 · 稀疏Token选择
University of Illinois at Urbana-Champaign · Microsoft Research · Google DeepMind
用单个可学习Token从视觉KV缓存中检索稀疏相关token,大幅提升长视觉上下文中的问答准确率
📌 工作参考价值(4 点)
- 电商场景中处理商品多图或视频描述时,可利用可学习检索Token从大量视觉特征中快速定位与用户查询相关的帧/区域,提升问答或推荐理由生成的质量。
- 利用预填充的KV缓存和稀疏选择机制,避免每次查询都重新处理所有视觉token,大幅降低多模态推荐的推理延迟和显存占用,适合线上服务。
- 轻量级设计(单个token,单卡训练)表明,小型Query-Image数据集即可训练出有效的检索能力,可迁移到商品视觉问答或视觉搜索中。
- 该方法本质是条件特征选择,可扩展到用户行为序列(如视频浏览记录)中,用学习到的Token从长序列中检索相关历史行为,增强推荐模型而无需输入完整序列。